package cn.unknowpupil.init.components.aspect;

import cn.unknowpupil.init.components.annotation.RunTime;
import lombok.extern.slf4j.Slf4j;
import org.aspectj.lang.ProceedingJoinPoint;
import org.aspectj.lang.annotation.Around;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.annotation.Pointcut;
import org.aspectj.lang.reflect.MethodSignature;
import org.springframework.stereotype.Component;

import java.lang.reflect.Method;

/**
 * Created at 2020/10/22 by w10g <br>
 * Initialize RunTimeAspect ... <br>
 *
 * @author unknowpupil
 * @since 0.0.1
 */
@Slf4j
@Aspect
@Component
public class RunTimeAspect {
    // 一秒钟，即1000ms
    private static final long ONE_MINUTE = 1000;

    // service层的统计耗时切面，类型必须为final String类型的,注解里要使用的变量只能是静态常量类型的
    @Pointcut("@annotation(cn.unknowpupil.init.components.annotation.RunTime)")
    public void pointcut() {
    }

    /**
     * 统计方法执行耗时Around环绕通知
     * @param joinPoint
     * @return
     */
    @Around("pointcut()")
    public Object timeAround(ProceedingJoinPoint joinPoint) {

        // 定义返回对象、得到方法需要的参数
        Object obj = null;
        Object[] args = joinPoint.getArgs();
        long startTime = System.currentTimeMillis();

        try {
            obj = joinPoint.proceed(args);
        } catch (Throwable e) {
            log.error("统计某方法执行耗时环绕通知出错", e);
        }

        // 获取执行的方法名
        long endTime = System.currentTimeMillis();
        MethodSignature signature = (MethodSignature) joinPoint.getSignature();
        String methodName = signature.getDeclaringTypeName() + "." + signature.getName();
        Method signatureMethod = signature.getMethod();
        RunTime limit = signatureMethod.getAnnotation(RunTime.class);
        String describe = limit.describe();
        // 打印耗时的信息
        this.printExecTime(describe,methodName, startTime, endTime);

        return obj;
    }
    /**
     * 打印方法执行耗时的信息，如果超过了一定的时间，才打印
     * @param methodName
     * @param startTime
     * @param endTime
     */
    private void printExecTime(String describe ,String methodName, long startTime, long endTime) {
        long diffTime = endTime - startTime;
        if (diffTime > ONE_MINUTE) {
            log.warn("{} 大于一秒钟-----{} 方法执行耗时：{} ms",describe,methodName,diffTime);
        }else {
            log.info("{} 小于一秒钟-----{} 方法执行耗时：{} ms",describe,methodName,diffTime);
        }
    }
}
